package com.extension.company.common.widget;

import android.content.Context;
import android.content.res.TypedArray;
import android.os.Build;
import android.support.annotation.NonNull;
import android.support.annotation.Nullable;
import android.support.annotation.RequiresApi;
import android.support.design.widget.TabLayout;
import android.support.v4.view.ViewPager;
import android.util.AttributeSet;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.FrameLayout;
import android.widget.TextView;

import com.extension.company.admin.R;

import java.lang.ref.WeakReference;
import java.util.ArrayList;
import java.util.List;

/**
 * 对 support Design 包中的TabLayout包装
 * 主要实现功能：更改indicator 的长度
 * Created by zhouwei on 2018/5/18.
 */

public class EnhanceTabLayout extends FrameLayout {
	private TabLayout mTabLayout;
	private List<String> mTabList;
	private List<View> mCustomViewList;
	private int mSelectIndicatorColor;
	private int mSelectTextColor;
	private int mUnSelectTextColor;
	private int mIndicatorHeight;
	private int mIndicatorWidth;
	private int mTabMode;
	private int mTabTextSize;

	public EnhanceTabLayout(@NonNull Context context) {
		super(context);
		init(context, null);
	}

	public EnhanceTabLayout(@NonNull Context context, @Nullable AttributeSet attrs) {
		super(context, attrs);
		init(context, attrs);
	}

	public EnhanceTabLayout(@NonNull Context context, @Nullable AttributeSet attrs, int defStyleAttr) {
		super(context, attrs, defStyleAttr);
		init(context, attrs);
	}

	@RequiresApi(api = Build.VERSION_CODES.LOLLIPOP)
	public EnhanceTabLayout(@NonNull Context context, @Nullable AttributeSet attrs, int defStyleAttr, int defStyleRes) {
		super(context, attrs, defStyleAttr, defStyleRes);
		init(context, attrs);
	}

	private void readAttr(Context context, AttributeSet attrs) {
		TypedArray typedArray = context.obtainStyledAttributes(attrs, R.styleable.EnhanceTabLayout);
		mSelectIndicatorColor = typedArray.getColor(R.styleable.EnhanceTabLayout_tabIndicatorColor, context.getResources().getColor(R.color.colorAccent));
		mUnSelectTextColor = typedArray.getColor(R.styleable.EnhanceTabLayout_tabTextColor, context.getResources().getColor(R.color.black)); //Color.parseColor("#666666")
		mSelectTextColor = typedArray.getColor(R.styleable.EnhanceTabLayout_tabSelectTextColor, context.getResources().getColor(R.color.colorAccent));
		mIndicatorHeight = typedArray.getDimensionPixelSize(R.styleable.EnhanceTabLayout_tabIndicatorHeight, 1);
		mIndicatorWidth = typedArray.getDimensionPixelSize(R.styleable.EnhanceTabLayout_tabIndicatorWidth, 0);
		mTabTextSize = typedArray.getDimensionPixelSize(R.styleable.EnhanceTabLayout_tabTextSize, 13);
		mTabMode = typedArray.getInt(R.styleable.EnhanceTabLayout_tab_Mode, 2);
		typedArray.recycle();
	}

	private void init(Context context, AttributeSet attrs) {
		readAttr(context, attrs);

		mTabList = new ArrayList<>();
		mCustomViewList = new ArrayList<>();
		View view = LayoutInflater.from(getContext()).inflate(R.layout.enhance_tab_layout, this, true);
		mTabLayout = view.findViewById(R.id.enhance_tab_view);

		// 添加属性
		mTabLayout.setTabMode(mTabMode == 1 ? TabLayout.MODE_FIXED : TabLayout.MODE_SCROLLABLE);
		mTabLayout.addOnTabSelectedListener(new TabLayout.OnTabSelectedListener() {
			@Override
			public void onTabSelected(TabLayout.Tab tab) {
				// onTabItemSelected(tab.getPosition());
				// Tab 选中之后，改变各个Tab的状态
				for (int i = 0; i < mTabLayout.getTabCount(); i++) {
					View view = mTabLayout.getTabAt(i).getCustomView();
					if (view == null) {
						return;
					}
					TextView text = (TextView) view.findViewById(R.id.tab_item_text);
					View indicator = view.findViewById(R.id.tab_item_indicator);
					if (i == tab.getPosition()) { // 选中状态
						text.setTextColor(mSelectTextColor);
						indicator.setBackgroundColor(mSelectIndicatorColor);
						indicator.setVisibility(View.VISIBLE);
					} else {// 未选中状态
						text.setTextColor(mUnSelectTextColor);
						indicator.setVisibility(View.INVISIBLE);
					}
				}
			}

			@Override
			public void onTabUnselected(TabLayout.Tab tab) {

			}

			@Override
			public void onTabReselected(TabLayout.Tab tab) {

			}
		});
	}

	public List<View> getCustomViewList() {
		return mCustomViewList;
	}

	public void addOnTabSelectedListener(TabLayout.OnTabSelectedListener onTabSelectedListener) {
		mTabLayout.addOnTabSelectedListener(onTabSelectedListener);
	}

	/**
	 * 与TabLayout 联动
	 *
	 * @param viewPager
	 */
	public void setupWithViewPager(@Nullable ViewPager viewPager) {
		mTabLayout.addOnTabSelectedListener(new ViewPagerOnTabSelectedListener(viewPager, this));
	}


	/**
	 * retrive TabLayout Instance
	 *
	 * @return
	 */
	public TabLayout getTabLayout() {
		return mTabLayout;
	}

	/**
	 * 添加tab
	 *
	 * @param tab tablayout的标题
	 */
	public void addTab(String tab) {
		mTabList.add(tab);
		View customView = getTabView(getContext(), tab, mIndicatorWidth, mIndicatorHeight, mTabTextSize,mUnSelectTextColor);
		mCustomViewList.add(customView);
		mTabLayout.addTab(mTabLayout.newTab().setCustomView(customView));
	}

	public static class ViewPagerOnTabSelectedListener implements TabLayout.OnTabSelectedListener {

		private final ViewPager mViewPager;
		private final WeakReference<EnhanceTabLayout> mTabLayoutRef;

		public ViewPagerOnTabSelectedListener(ViewPager viewPager, EnhanceTabLayout enhanceTabLayout) {
			mViewPager = viewPager;
			mTabLayoutRef = new WeakReference<EnhanceTabLayout>(enhanceTabLayout);
		}

		@Override
		public void onTabSelected(TabLayout.Tab tab) {
			mViewPager.setCurrentItem(tab.getPosition());
			EnhanceTabLayout mTabLayout = mTabLayoutRef.get();
			if (mTabLayoutRef != null) {
				List<View> customViewList = mTabLayout.getCustomViewList();
				if (customViewList == null || customViewList.size() == 0) {
					return;
				}
				for (int i = 0; i < customViewList.size(); i++) {
					View view = customViewList.get(i);
					if (view == null) {
						return;
					}
					TextView text = (TextView) view.findViewById(R.id.tab_item_text);
					View indicator = view.findViewById(R.id.tab_item_indicator);
					if (i == tab.getPosition()) { // 选中状态
						text.setTextColor(mTabLayout.mSelectTextColor);
						indicator.setBackgroundColor(mTabLayout.mSelectIndicatorColor);
						indicator.setVisibility(View.VISIBLE);
					} else {// 未选中状态
						text.setTextColor(mTabLayout.mUnSelectTextColor);
						indicator.setVisibility(View.INVISIBLE);
					}
				}
			}

		}

		@Override
		public void onTabUnselected(TabLayout.Tab tab) {
			// No-op
		}

		@Override
		public void onTabReselected(TabLayout.Tab tab) {
			// No-op
		}
	}

	/**
	 * 获取Tab 显示的内容
	 *
	 * @param context
	 * @param text            tablayout 标题的文字
	 * @param indicatorWidth  指示器的宽度
	 * @param indicatorHeight 指示器的高度
	 * @param textSize        标题文字的大小
	 * @param mUnSelectTextColor 未选中的字体默认的颜色
	 * @return
	 */
	public static View getTabView(Context context, String text, int indicatorWidth, int indicatorHeight, int textSize,int mUnSelectTextColor) {
		View view = LayoutInflater.from(context).inflate(R.layout.tab_item_layout, null);  // tablayot 的item View
		TextView tabText = (TextView) view.findViewById(R.id.tab_item_text);    // tablayout item 标题
		if (indicatorWidth > 0) {
			View indicator = view.findViewById(R.id.tab_item_indicator);
			ViewGroup.LayoutParams layoutParams = indicator.getLayoutParams();
			layoutParams.width = indicatorWidth;
			layoutParams.height = indicatorHeight;
			indicator.setLayoutParams(layoutParams);
		}
		tabText.setTextSize(textSize);
		tabText.setText(text);
		tabText.setTextColor(mUnSelectTextColor);
		return view;
	}
}
